package com.zhuang.tree;

public class ArrBinaryTreeDemo {
    public static void main(String[] args) {
        int[] arr = {1, 2, 3, 4, 5, 6, 7};
        ArrBinaryTree arrBinaryTree = new ArrBinaryTree(arr);
        System.out.println("前序遍历");
        arrBinaryTree.perOrder();
        System.out.println("中序遍历");
        arrBinaryTree.infixOrder();
        System.out.println("后序遍历");
        arrBinaryTree.postOrder();
    }
}

class ArrBinaryTree {
    // 存储数据节点的数组
    private int[] arr;

    public ArrBinaryTree(int[] arr) {
        this.arr = arr;
    }

    // 重载一个方法
    public void perOrder() {
        this.perOrder(0);
    }

    public void infixOrder() {
        this.infixOrder(0);
    }

    public void postOrder() {
        this.postOrder(0);
    }

    /**
     * 二叉树的前序遍历
     *
     * @param index 数组的下标
     */
    public void perOrder(int index) {
        // 如果数组为空 或者 arr.lenght=0
        if (arr == null || arr.length == 0) {
            System.out.println("数组为空，不能二叉树前序遍历");
        }
        // 输出当前元素
        System.out.println(arr[index]);
        // 向左递归遍历
        if ((2 * index + 1) < arr.length) {
            perOrder(2 * index + 1);
        }
        // 向右递归遍历
        if ((2 * index + 2) < arr.length) {
            perOrder(2 * index + 2);
        }
    }


    /**
     * 二叉树的中序遍历
     *
     * @param index 数组的下标
     */
    public void infixOrder(int index) {
        // 如果数组为空 或者 arr.lenght=0
        if (arr == null || arr.length == 0) {
            System.out.println("数组为空，不能二叉树中序遍历");
        }
        // 向左递归遍历
        if ((2 * index + 1) < arr.length) {
            infixOrder(2 * index + 1);
        }
        // 输出当前元素
        System.out.println(arr[index]);
        // 向右递归遍历
        if ((2 * index + 2) < arr.length) {
            infixOrder(2 * index + 2);
        }
    }

    /**
     * 二叉树的后序遍历
     *
     * @param index 数组的下标
     */
    public void postOrder(int index) {
        // 如果数组为空 或者 arr.lenght=0
        if (arr == null || arr.length == 0) {
            System.out.println("数组为空，不能二叉树后序遍历");
        }
        // 向左递归遍历
        if ((2 * index + 1) < arr.length) {
            postOrder(2 * index + 1);
        }
        // 向右递归遍历
        if ((2 * index + 2) < arr.length) {
            postOrder(2 * index + 2);
        }
        // 输出当前元素
        System.out.println(arr[index]);
    }
}
